package com.codingxyz.simple.prodconsumer;


import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

class MyResource{

    private volatile boolean FLAG = true;//默认开启。进行生产消费活动

    private AtomicInteger atomicInteger = new AtomicInteger();

    BlockingQueue<String> blockingQueue = null;

    public MyResource(BlockingQueue<String> blockingQueue) {
        this.blockingQueue = blockingQueue;
        System.out.println(blockingQueue.getClass().getName());
    }

    public void myprod() throws Exception{

        String data = null;
        boolean retValue;

        while (FLAG){
            data = atomicInteger.incrementAndGet()+"";
            retValue = blockingQueue.offer(data,2L, TimeUnit.SECONDS);

            if(retValue){
                System.out.println(Thread.currentThread().getName() + " \t 插入队列 "+data+"成功");
            }else{
                System.out.println(Thread.currentThread().getName() + " \t 插入队列 "+data+"失败");
            }

            TimeUnit.SECONDS.sleep(1);
        }

        System.out.println(Thread.currentThread().getName() + " \t 大老板叫停了，表示FLAG=false,生产结束");

    }


    public void  myconsumer() throws Exception{

        String result = null;
        while (FLAG){

            result = blockingQueue.poll(2L,TimeUnit.SECONDS);
            if(null == result || result.equalsIgnoreCase("")){
                FLAG = false;
                System.out.println(Thread.currentThread().getName() + " \t 超过2s没有取到数据，消费者退出 ");

                System.out.println();
                System.out.println();
                System.out.println();
                return;
            }

            System.out.println(Thread.currentThread().getName() + " \t 消费者队列退出 "+result+" 成功");
        }


    }


    public void stop(){
        FLAG =false;
    }


}


public class ProdConsumer_BlockQueueDemo {


    public static void main(String[] args) {

        MyResource myResource = new MyResource(new ArrayBlockingQueue<String>(10));

        System.out.println(Thread.currentThread().getName() + " \t 生产者线程启动 " );
        System.out.println();
        System.out.println();
        System.out.println();
        System.out.println();
        new Thread(()->{

            try {
                myResource.myprod();
            } catch (Exception e) {
                e.printStackTrace();
            }


        },"Prod ").start();


        new Thread(()->{

            System.out.println(Thread.currentThread().getName()+" \t 消费者线程启动 ");

            try {
                myResource.myconsumer();
            } catch (Exception e) {
                e.printStackTrace();
            }

        },"consumer ").start();

        try {
            TimeUnit.SECONDS.sleep(5);
        }catch (Exception e){
            e.printStackTrace();
        }

        System.out.println();
        System.out.println();
        System.out.println();

        System.out.println("5s中后，结束线程");

        myResource.stop();
    }


}
